first attempt at knn classification