Added mock test for mask, created method that would test it