A restart button for in the game