Added functionality to the game states.