Add in your code

#include <vector>
#include <string>