Added support for custom input dialog