- ib.arrow().rectangle(policy.getName(), RectTypes.POLICY, policy.getPolicyModel().getPolicyAcronym());
+ ib.arrow().rectangle(policy.getName(), RectTypes.POLICY, policy.getPolicyModel().getPolicyAcronym(),
+ policy.getLoopElementModel() != null ? policy.getLoopElementModel().getName() : null,
+ policy.getLoopElementModel() != null ? policy.getLoopElementModel().getName() : null);