Update ps5_mapreduce.py