Added mask tests
Add input tests