add to bayesian model too