Added rounded background to dialog box
Minor refactoring