updated kmeans to have a class