Implement KNN from Scratch | J.P. Morgan Coding Question