Add keyBy() to the RDD class.