Update atom.py

Feel free to change get_neighbors the if statements can be written in fewer lines