Added new features to MainWindow