Added more code to MapReduce