Changed from widget to MainWindow