Added edit form for treatments