Removed TreatmentShallowSerializer (not needed)
Adding tests for Treatment API