Fixed mask issue
Merge pull request #20 from Mitch528/master

Fixed mask issue