Add a modified SparkKMeans2 cas