Improve dialog box UI