Update treeGDHeat.py