Change test script to use Screen object