Few changes in UI if dialog box