Updating treestore.py