Add .sortBy(f) method on RDD