Updated test.go for transpose