Add restart framew