Added header files in CMake for project. Fix #7