updated get_mask
updated gen_mask