Change to use QDialogButtonBox