Added feature for saving images