Update Heat.py after staging
Heat.py has numpy arraysnow!