Update k-means.rb