Clean up, add ggsave() to save plots as images