make_square_grid function

Create a square grid