Update enrichment.py