Added kmeans code