Add restart handler.