add axis arg, write transposed array