Proper restart at end of game

instead of returning to the home screen