Feature/kmeans

Adding kmeans and some unit tests