Add support for data attributes for custom buttons (dialog)