Add end game state checking