How to change controls when changing stages