Added option to save the image