Update esi_calc.py