correct header include in main.cpp