Update tri_tree.py