} else if (context.getNewObject() instanceof Pool) {
return new AddPoolFeature(this);
} else if (context.getNewObject() instanceof Lane) {
return new AddLaneFeature(this);
} else if (context.getNewObject() instanceof CallActivity) {
return new AddCallActivityFeature(this);
} else if (context.getNewObject() instanceof TextAnnotation) {
return new AddTextAnnotationFeature(this);
}
return super.getAddFeature(context);
}