Update registration.py