Add option to restart the game