Improve the grid. Add missing  variable