Edit: masking.py