Update register.py

Add salt functionality