Minor changes to MainDialog