@Override
	public IDecorator[] getDecorators(PictogramElement pe) {

		if(pe instanceof ContainerShape && pe instanceof Diagram == false){

			Validator validator = new Validator(diagramTypeProvider);
			Object object = featureProvider.getBusinessObjectForPictogramElement(pe);

			if(object instanceof State){

				ValidationMessage message = validator.createWarnings((State)object);
				if(message != null){

					if(pe instanceof ContainerShape && pe instanceof Diagram == false){

						IImageDecorator imageRenderingDecorator = null;

						if(message.getSeverity().equals(ValidationSeverity.WARNING)){

							imageRenderingDecorator = new ImageDecorator(IPlatformImageConstants.IMG_ECLIPSE_WARNING_TSK);
						}

						else if(message.getSeverity().equals(ValidationSeverity.ERROR)){

							imageRenderingDecorator = new ImageDecorator(IPlatformImageConstants.IMG_ECLIPSE_ERROR_TSK);
						}

						else{

							return super.getDecorators(pe);
						}

						imageRenderingDecorator.setMessage(message.getMessage());
						imageRenderingDecorator.setX(pe.getGraphicsAlgorithm().getWidth()-20);
						return new IDecorator[] { imageRenderingDecorator };
					}
				}
			}
		}
		return super.getDecorators(pe);
	}

Shape newShape = peCreateService.createShape((ContainerShape) pe, false);
				Ellipse ellipse = gaService.createEllipse(newShape);
				ellipse.setForeground(manageColor(IColorConstant.BLACK));
				ellipse.setBackground(manageColor(IColorConstant.BLACK));
				ellipse.setLineWidth(1);
				gaService.setLocationAndSize(ellipse, xPos, yPos, 7, 7);

@Override
	protected void refreshVisuals() {
		super.refreshVisuals();
		refreshSourceConnections();
		refreshTargetConnections();
		delegate.refreshFigureForEditPart();
	}

protected void refreshChildren() {
		int i;
		EditPart editPart;
		Object model;

		List children = getChildren();
		int size = children.size();
		Map modelToEditPart = Collections.EMPTY_MAP;
		if (size > 0) {
			modelToEditPart = new HashMap(size);
			for (i = 0; i < size; i++) {
				editPart = (EditPart) children.get(i);
				modelToEditPart.put(editPart.getModel(), editPart);
			}
		}

		List modelObjects = getModelChildren();
		for (i = 0; i < modelObjects.size(); i++) {
			model = modelObjects.get(i);

			// Do a quick check to see if editPart[i] == model[i]
			if (i < children.size()
					&& ((EditPart) children.get(i)).getModel() == model)
				continue;

			// Look to see if the EditPart is already around but in the
			// wrong location
			editPart = (EditPart) modelToEditPart.get(model);

			if (editPart != null)
				reorderChild(editPart, i);
			else {
				// An EditPart for this model doesn't exist yet. Create and
				// insert one.
				editPart = createChild(model);
				addChild(editPart, i);
			}
		}