updated the restart method