Add support for visualizations