Add restart functionality.