Move the core loops into module-level helpers.

Not sure if it really matters, but it does make it a bit simpler, and it
means the code matches _update_from_first_parent.