Update DHE.py