Add gracephul restart