Some more options for plot/distribution_matrix

Still lots of things to improve. But should be good enough
for mcmc investiagtions, probably not good enough for papers.