Implemented missing restart function