added plots
plot images
removed an unused variable and changed the name of a function
modified plotting code