small progress on kmeans