Mask Function In Python