add game state handling