Change columnDimensions from (56, 56) to (56 * 56) to get rid of an
assertion error.