Update bayesian_network.py