Added a game state to make things a bit clearer