KNN visualization in just 13 lines of code