Create create_squared_grid.py