changes on main.cpp