updated hw1_knn file