first crack at completing assignment. need to clean up last plot and save pngs