Improve image generation to prevent text