Add more diagnostic plots for CLI

[thomasckng]

Currently only the final corner plot is outputted from the CLI. More diagnostic should be included for all samplers.