added setTreatment
-george