Add restart and save scripts