K-Nearest Neighbor classification using python