Pruning the tree
Minor changes