Added generate_tree.py