Update atom.py