changed treatment stuff a little