Add save_info to the task_spec