Accept different inputs when asking user to restart