Small change in mainWindow