Example of KNN with Scikit-Learn