Update main.cpp

enough for demo