Add missing headers and doc blocks