Update main_window.py