Updating the grids with the right classes