main.cpp fix