Update helper_functions.py