Add transpose + tests