Create mask.py