@Override
public boolean canCreate(ICreateConnectionContext context) {
		PictogramElement sourcePictogramElement = context.getSourcePictogramElement();
		PictogramElement targetPictogramElement = context.getTargetPictogramElement();

		if (getBusinessObjectForPictogramElement(sourcePictogramElement) instanceof AbstractState && getBusinessObjectForPictogramElement(targetPictogramElement) instanceof AbstractState) {
			targetState = (AbstractState) getBusinessObjectForPictogramElement(targetPictogramElement);
			sourceState = (AbstractState) getBusinessObjectForPictogramElement(sourcePictogramElement);
			return true;
		}
		
		return sourceState != null && targetState != null && sourcePictogramElement != null && targetPictogramElement != null;
	}

	@Override
	public Connection create(ICreateConnectionContext context) {
		Connection newConnection = null;
		
		final Transition transition = StatemachineFactory.eINSTANCE.createTransition();
		transition.setName("Transition");
		if (targetState != null) {
			transition.setState(targetState);
		}
		
		if(sourceState != null) {
                       // If I put any modification here, they will not be saved
		}
		
		AddConnectionContext addContext = new AddConnectionContext(context.getSourceAnchor(), context.getTargetAnchor());
		addContext.setNewObject(transition);
		newConnection = (Connection) getFeatureProvider().addIfPossible(addContext);
		
		getDiagram().eResource().getContents().add(transition);

		return newConnection;
	}