main.cpp updated