implement code to pass first temp TDD test for kmeans clustering