Simple Bayesian Network in python