working on adding vector on main.cpp