Adding a function to plot the neighborhood