Made the game rest controlled by a button