I think stage 2 looks pretty great and by the end, it'll be this in-line progression as things get cleaner, tidier and more practiced.

Good luck and you got this!!! Take care of yourself and be patient with yourself during recovery. <3