I like your ideas. Maybe some extra UI after picking your upgrade? Like two buttons on the bottom right of the screen: one to teleport back to the middle and another one to start the next round?