notebook for KMeans