Fix docstring in distance.py