Minor tweaks to visualizations and code cleanup.