Update mainwindow.py