Add condition to end the game when it's over