modify ajust_tree.py