Adding code for visualizations