Adding test for layout