Widow/Orphan Control runs in Word by default. However, it doesn’t always seem like it. While it keeps single lines of a paragraph from showing up either at the top or the bottom of a page, it doesn’t necessarily keep headings with the paragraphs they describe or diagrams with the text that refers to them.
Select the heading or diagram that’s being orphaned or widowed, then the dialog box launcher in the Paragraph group. Select the Line and Page Breaks tab and click “Keep with next” and “Keep lines together.” Choosing both will ensure things stay together as your document changes.