added models TreatmentType and TreatmentName