Update atom-selection.ipynb