Update write_lookup.py