Add simple kNN code