United Airlines interview question

write knn code from scratch