Update numpy_map.py