Add register.py