It would be helpful to be able to select the number of columns as well as the number of rows when creating the stage flow.