Update es_util.py