Fixing bug in grid.rs example