Improving the tree!