Add restart! method