Update mask.py