Made changes to dialog box