Change game over check to be more extensible