Improve docs in mps_gen.
Improve docs for tdvp_gen.