Adds more tests for kmeans