Added image generation