Update main.cpp

is needed