Better Kmeans loop for any number of clusters