Update image_transform.py