Add another function to generate the grid. Create generate_grid.