Add tree.py