Add endgame check