corrections to plots for knn