V1 of knn_visualization