Add standard comments to mask.py