Update tree-initialize.py