Comments and tweak.
Only giving a reind_arr when it is needed.

A bit more complicated, but logically more satisfactory, and paves the
way for fewer matrix exponentiations.

@nhoffman-- Could you give this one a spin? I'd be interested in a speed
comparison (though that is of course secondary).