Add dimnames to output
Fix for censored