modified enrich.py