Add test for getting columns from header