Update main_2.cpp