add back in the image creation