add restart method