add mask.py