Add TreeStore.py