Upgrades to pdiag for improved distribution diagnostics