Added unit test for Grid class