Small change to persist the image after being taken