Update reg.py