Improved UI of DialogWidget