Changes in MainDialog class