Added saving of current image