Update main_class.cc