Add restarting/quitting when the game is over