Update register.py