Add getWeightMask()