Added restart function for end-game and game menu.

Needs testing.