Update main.cpp