add image generation