improved list comprehension for neighbors