fix traceplot for discrete
use proper logp