Added training and test files and updated kNN to use them