Reinstate restart function