add restart function and test