Create tree.py