Added restart function