Add game state and menu screen