@Override
	public PictogramElement add(IAddContext context) {
		ClassObject addedClass = (ClassObject) context.getNewObject();
		Diagram targetDiagram = (Diagram) context.getTargetContainer();

		IPeCreateService peCreateService = Graphiti.getPeCreateService();
		IGaService gaService = Graphiti.getGaService();

		/***********/
		/** CLASS **/
		/***********/
		ContainerShape classContainer = peCreateService.createContainerShape(
				targetDiagram, true);

		// check whether the context has a size (e.g. from a create feature)
		// otherwise define a default size for the shape
		final int width = context.getWidth() <= 0 ? 100 : context.getWidth();
		final int height = context.getHeight() <= 0 ? 200 : context.getHeight();

		// create and set graphics algorithm
		Rectangle roundedRectangle = gaService.createRectangle(classContainer);
		roundedRectangle.setForeground(manageColor(CLASSOBJECT_FOREGROUND));
		roundedRectangle.setBackground(manageColor(CLASSOBJECT_BACKGROUND));
		roundedRectangle.setLineWidth(1);
		gaService.setLocationAndSize(roundedRectangle, context.getX(),
				context.getY(), width, height);

		// create link and wire it
		link(classContainer, addedClass);

		/**********/
		/** LINE **/
		/**********/
		// create shape for line
		Shape shape = peCreateService.createShape(classContainer, false);

		// create and set graphics algorithm
		Polyline polyline = gaService.createPolyline(shape, new int[] { 0,
				LINE_HEIGHT, width, 20 });
		polyline.setForeground(manageColor(CLASSOBJECT_FOREGROUND));
		polyline.setLineWidth(2);

		/**********************************/
		/** CLASS NAME (shape with text) **/
		/**********************************/
		// create shape for text
		Shape classNameShape = peCreateService.createShape(classContainer,
				false);

		// create and set text graphics algorithm
		Text className = gaService.createText(classNameShape,
				addedClass.getName());
		className.setForeground(manageColor(CLASSOBJECT_TEXT_FOREGROUND));
		className.setHorizontalAlignment(Orientation.ALIGNMENT_CENTER);
		// vertical alignment has as default value "center"
		className.setFont(gaService
				.manageDefaultFont(getDiagram(), false, true));
		gaService.setLocationAndSize(className, 0, 0, width,
				DEFAULT_HEIGHT_TEXT);

		// create link and wire it
		link(classNameShape, addedClass);

		/*****************/
		/** ATTRIBUTES **/
		/*****************/
                // just for testing
		Attribute attribute = new Attribute("TestA1");

		ContainerShape attributeContainer = peCreateService
				.createContainerShape(classContainer, true);

		// create and set graphics algorithm
		Rectangle attribRectangle = gaService
				.createInvisibleRectangle(attributeContainer);

		attribRectangle.setTransparency(0.0);
		attribRectangle.setForeground(manageColor(CLASSOBJECT_FOREGROUND));
		attribRectangle.setBackground(manageColor(CLASSOBJECT_BACKGROUND));
		attribRectangle.setLineWidth(1);
		gaService.setLocationAndSize(attribRectangle, 0, LINE_HEIGHT, width,
				height);

		// Create an add context that describes where to add the Attribute in
		// the class
		AddContext addAttribContext = new AddContext(new AreaContext(),
				attribute);
		addAttribContext.setWidth(width);

		addAttribContext.setTargetContainer(attributeContainer);

		// Trigger the Graphiti framework to find a feature that is capable of
		// adding a Attribute
		addGraphicalRepresentation(addAttribContext, attribute);

		// call the layout feature
		layoutPictogramElement(classContainer);

		return classContainer;
	}

	@Override
	public PictogramElement add(IAddContext context) {
		// Get information from context
		Attribute attrib = (Attribute) context.getNewObject();
		ContainerShape containerShape = context.getTargetContainer();

		IPeService peCreateService = Graphiti.getPeService();
		IGaService gaService = Graphiti.getGaService();

		// create shape for text
		Shape attribNameShape = peCreateService.createShape(containerShape,
				false);

		// create and set text graphics algorithm
		Text attribName = gaService.createText(attribNameShape,
				attrib.getName());
		attribName.setForeground(manageColor(CLASSOBJECT_TEXT_FOREGROUND));
		attribName.setHorizontalAlignment(Orientation.ALIGNMENT_LEFT);
		attribName.setVerticalAlignment(Orientation.ALIGNMENT_TOP);
		attribName.setFont(gaService.manageDefaultFont(getDiagram(), false,
				true));
		gaService.setLocationAndSize(attribName, 0, 0, context.getWidth(), 20);

		// create link and wire it
		link(attribNameShape, attrib);

		return attribNameShape;
	}

	Shape[] toDelete = oldAttributeCon.getChildren().toArray(
		       new Shape[oldAttributeCon.getChildren().size()]);
	for (Shape child : toDelete) {
		EcoreUtil.delete(child, true);
	}	

ContainerShape getFirstContainerShape()
ContainerShape setFirstContainerShape()
public Shape getNameShape()
public String getName()
public void setName(String name)
...etc...

* If the meaning of the ... isn't clear,
* there really should be more of a description here...

- container : rectangle
-- shape : polyline
-- shape : text (for classname)
-- container : rectangle (for attributes)
--- shapes : text (for each attribute)

public ClassObjectUtil(ContainerShape classContainer) {
		this.classContainer = classContainer;

		// set all other shapes we know
		for (Shape child : classContainer.getChildren()) {
			if (child instanceof ContainerShape) {
				attributeContainer = (ContainerShape) child;
			} else if (child instanceof Shape) {
				// could be name or line
				GraphicsAlgorithm ga = child.getGraphicsAlgorithm();
				if (ga instanceof Text) {
					classNameShape = child;
				} else if (ga instanceof Polyline) {
					line = (Polyline) ga;
				}
			}
		}
	}

	public String getClassName() {
		return ((Text) classNameShape.getGraphicsAlgorithm()).getValue();
	}

...
...
...