Add restart grain functionality