Adds tests for custom layout