Add restart to gameplay