Working on main.cpp