Update main.cpp

Remove vector header.