Improved `quick_visualizations.py`.