Add arg to grid function for number of rows.