Trying mapPartitions