Refined the integration of the visualizations code and the visualizer