modify TREE.py