Add restart step