cleaned up code for StartScreen and gameStates