Add restart handlers