Update aggregation.py