updated custom.py