kmeans working. Need to save the clusters still