Add support for dialogs