added code for treatments