Update findNeighborhood function