refactor screen mask