Knn Code In Python Using Return