Fix kmeans_2d example