Update map_reduce.py