Update main_function.py