Add a with_neighbors helper method