Update custom.py

Added blank line before function definition