Changes to edit and add on new tree