Update callback_functions.py