Update main.cpp

Небольшие доработки