Minor edit to add header files to project