Update heimdall.py