Update treeBuild.py