Create tree_generation.py