Adding Mask class