Merging of data from branch_save_treatment_factor

Code review : Daniel Jao