The original implementation accepted an array which infact was just an integer cast as
np.ndarray for each element of which (in this case 1),
m was calculated which in turn was used to calculate the distance, which again was an
I've removed an unnecessary variable
size and the implementation now directly uses the 'index' provided. The results are identical as the original implementation.