Create get_mask.py