Update image_registration.py