axis2: axis 1 is out of bounds for array of dimension 1
axis2: axis 1 is out of bounds for array of dimension 1