Update main_with_callback.py