Updated the visualization module