KNN first attempt