Added visualizations to fourth notebook