Update u_main.cpp