add restart capability