Add a modified SparkKMeans2 case