Wrote restart function
Fix merge