Update main.cpp

Arg # fix.