Added a button to restart the game