Updates to K-means