Update demo_grid.m