Changed to KMeans from sklearn instead of scipy