Tweaking the UI and added dialog box for backbutton