Update enrich.py