Update knn.go