Implement KNN from scratch | Other Interview Question