use new treatment API