modified main.cpp, header.h