Add support for dialog element