Add nicer menu screen and level select state (not yet implemented)