Add restart rules