Add test for new header class