Add remove support in tree