Improve styling for dialogs