Add test for grid