Updated the clustering algorithm to use k-means++