Update knn.rb