Update header file for documentation (#2)