Added some test code for the grid