Update main.cpp

add "yes" string