Add a little GameState logic