Add mask enable to diagram