Removing Mask - epoch 4