minor changes in KMeans