Safety :)
CDBN version of the visualizer
Fix the shape