Update main_ui.py