is this supposed to be "Unit" or RDD? (end of the function does a toRdd)