Small improvement to create_screen_wall