Refactoring neighbors function to utilize list comprehensions instead of for loops
Note: Seems to be much faster