Enabling restart of the game in the middle

This was the task 1 and task 2 of assignment 2.