Updata the main.cpp