Update main.cpp

add main func in main file