update entity_aggregation_toolkit.py