updates for tree.py and treeDriver.py