Modify dialog UI