Update salt-custom.py