Add restart hooks