added src and header. Also changed CMake