add tree.py