Fixing the "play again" feature

Also fixed the "quit" feature