I think extend getButton method available for CustomDialogLayout to Others dialogs (Input dialog) can be a good idea. Also i ask if is possibile to add a getTitle method (like getButton) to get Label used for title. This helps us to editing title without to add a custom label inside the dialog panel