treatment_code function

Use treatment coding