Update HEAT_Tools.py