implement the kmeans using sklearn