Updated doc of TreatmentParams
Updated TODO