finally pushing my updated KNN implentation