update register.py