first draft of full assignment, need to add better plots with ggplot