Correcting an error on the nodes distribution function
Adding square lattice network