Update gen_map.py