Added restart function for end-game and game menu.