Update new_main.cpp