treatment_dict implemented for plots