Changes in the tree generation tool