Added test for the custom header functionality