Updated Cells & gen_grid for 2d lattices