Update main.cpp

(writing the task)