Restart function added