Adding image directory for grabbing images
change name of plots
Removing useless code