Add missing headers
(issue #44)