Add option to specify mask