Update he_ex.py

updated