Create draw_tree.py