Add tests for custom headers