Restart_game function fixed and working