Updates to Dialog UI