Adding (2) code for the grid