Update treatment.java