added visualizations and code

Also, note that there is a ggsave function that makes it easy to save plots as images.