reshaping now only needs the number of rows