Update main.cpp

Update main.cpp #19