added plotting code and png's