Clustering data points using the KMeans algorithm