Update headers to work with flatdoc better