Cleaning up kmeans