Update kmeans demo