Add kmeans data generate code