Create a generic edit_selected button and push the "what to do when we edit" into the tree view, since it's usually view specific anyway.
Replace existing edit button with the new EditSelectedButton.