Update distance.py