joinWithVertices updated to take DataSet<Tuple2<K, T>> as input