Added mask method