adding tree.py