Update drawTree.py