Update main.cpp

Add little more.