Added header files to main.cpp.