Wrote docstring for euclidean_distances_argmin