Move dialog box methods to separate class