Improved formatting on output psi matrices