adding code and pngs for plotting project