Add tests for layouts