Visualization code add