Update create_image.py