Add BayesNetwork class to CreateNetwork component