improving the bayesian nodes