Fixed kmeans demo