Updated include directive in main.cpp