added inputs for specifying shape and number of cells to render