Update main.cpp

Small commit 2