Updating the grid class and example