Extract _find_atoms_ex
Use _find_atoms_ex in process()