modify MainWindow.py