Added missing header files in CMakeLists.txt