Update main.cpp

I fudge up on the code.