Checkpoint. Transpose for 3+d not working