Add tests for layout