game works but want to fix making the game end, starting a new one