new function to keep all chains in PT

for plotting densities between chains