Version 0.0.2 changes, adding dialog box