Adding options to save the plots