Better way to add classNames to dialogs