Add restart functionality