Add to kmeans demo