KMeans notebook script final