adding restart ability