Update TREE

Add a simpler TREE solution albeit less general