Update generator_custom_functions.py