Add tests of Screen module