Update heat2d.py