Yes, exactly. Once you solve the “puzzle” you begin to have the option to “stay” at the end of each life— then it’s up to you to decide which life you want to keep.