Working on saving edited treatments