Update atom.py

add calc_distance