Made changes to Kmeans