Changed class MainDialog