Machine Learning Interview - Implement the KNN Algorithm (with Snapchat MLE)