Update heat-transfer.py