Semi-heavily updating to handle properly ending the game.