Add tests for header