the code used for my KNN baseline submission