could we just add the class to the dialog div?