Updated doc and restructured Treatment and TreatmentParams for improved implementation.