Adding all the DBN stuff and visualizations