starting to implement kmeans