minor changes
KMeans