added restart to win&loss screens