Add the bayes diagram