Update KMeans.scala