Added Kmeans code