Add some game state handling