Update main.cpp

modify command line arguments