implement axes argument in transpose