Add kmeans test using the same data