main.cpp changes