Refactor restart of game functionality