kNN partially implemented