Add test for layout