Added implementation for visualizations