trace_cat function

Trace diagnostics for categorical randomizations