Add KNN plots
Test set