modified header.h, main.cpp