Updated k-means and driver code.