Changed the semantics of index (in mapPartitionsWithIndex) from output partition index (as defined in TaskContext) back to RDD partition index.

Also removed the extra MapPartitionsWIthContextRDD and put all the mapPartitions related functionality in MapPartitionsRDD.