Мелкие изменения в main.cpp