Add an explicit return type for this (I guess it will also be `RDD[T]`?)