Improve MainWindow state save/restore