Update distancetoatom.py