REMINDER - Change from QMainWindow to QWidget