Implementation of knn algorithm using sci-kit learn in Python