diff --git a/aiml/org.eclipse.papyrus.aiml.examples/models/kerasmodel.notation b/aiml/org.eclipse.papyrus.aiml.examples/models/kerasmodel.notation
index baee8e9..d8c6e72 100644
--- a/aiml/org.eclipse.papyrus.aiml.examples/models/kerasmodel.notation
+++ b/aiml/org.eclipse.papyrus.aiml.examples/models/kerasmodel.notation
@@ -77,13 +77,13 @@
       <element xmi:type="uml:Class" href="kerasmodel.uml#_wkcRgJwiEeqtcKkMgfhA1A"/>
       <layoutConstraint xmi:type="notation:Bounds" xmi:id="_gCKC4bM3Eeqmquzwsvk0ug" x="77" y="40" width="564" height="321"/>
     </children>
-    <children xmi:type="notation:Shape" xmi:id="_gCRXpbM3Eeqmquzwsvk0ug" type="StereotypeComment">
-      <styles xmi:type="notation:TitleStyle" xmi:id="_gCRXprM3Eeqmquzwsvk0ug"/>
-      <styles xmi:type="notation:EObjectValueStyle" xmi:id="_gCRXqLM3Eeqmquzwsvk0ug" name="BASE_ELEMENT">
+    <children xmi:type="notation:Shape" xmi:id="_j3V5YE_zEeuEueoCkMrBkg" type="StereotypeComment">
+      <styles xmi:type="notation:TitleStyle" xmi:id="_j3V5YU_zEeuEueoCkMrBkg"/>
+      <styles xmi:type="notation:EObjectValueStyle" xmi:id="_j3WgcE_zEeuEueoCkMrBkg" name="BASE_ELEMENT">
         <eObjectValue xmi:type="uml:Class" href="kerasmodel.uml#_wkcRgJwiEeqtcKkMgfhA1A"/>
       </styles>
       <element xsi:nil="true"/>
-      <layoutConstraint xmi:type="notation:Bounds" xmi:id="_gCRXp7M3Eeqmquzwsvk0ug" x="200"/>
+      <layoutConstraint xmi:type="notation:Bounds" xmi:id="_j3V5Yk_zEeuEueoCkMrBkg" x="277" y="40"/>
     </children>
     <styles xmi:type="notation:StringValueStyle" xmi:id="_fUr44bM3Eeqmquzwsvk0ug" name="diagram_compatibility_version" stringValue="1.4.0"/>
     <styles xmi:type="notation:DiagramStyle" xmi:id="_fUr44rM3Eeqmquzwsvk0ug"/>
@@ -91,15 +91,15 @@
       <owner xmi:type="uml:Model" href="kerasmodel.uml#_sfXSgJwiEeqtcKkMgfhA1A"/>
     </styles>
     <element xmi:type="uml:Model" href="kerasmodel.uml#_sfXSgJwiEeqtcKkMgfhA1A"/>
-    <edges xmi:type="notation:Connector" xmi:id="_gCR-sLM3Eeqmquzwsvk0ug" type="StereotypeCommentLink" source="_gCKC4LM3Eeqmquzwsvk0ug" target="_gCRXpbM3Eeqmquzwsvk0ug">
-      <styles xmi:type="notation:FontStyle" xmi:id="_gCR-sbM3Eeqmquzwsvk0ug"/>
-      <styles xmi:type="notation:EObjectValueStyle" xmi:id="_gCR-tbM3Eeqmquzwsvk0ug" name="BASE_ELEMENT">
+    <edges xmi:type="notation:Connector" xmi:id="_j3WgcU_zEeuEueoCkMrBkg" type="StereotypeCommentLink" source="_gCKC4LM3Eeqmquzwsvk0ug" target="_j3V5YE_zEeuEueoCkMrBkg">
+      <styles xmi:type="notation:FontStyle" xmi:id="_j3Wgck_zEeuEueoCkMrBkg"/>
+      <styles xmi:type="notation:EObjectValueStyle" xmi:id="_j3XHgU_zEeuEueoCkMrBkg" name="BASE_ELEMENT">
         <eObjectValue xmi:type="uml:Class" href="kerasmodel.uml#_wkcRgJwiEeqtcKkMgfhA1A"/>
       </styles>
       <element xsi:nil="true"/>
-      <bendpoints xmi:type="notation:RelativeBendpoints" xmi:id="_gCR-srM3Eeqmquzwsvk0ug" points="[0, 0, 0, 0]$[0, 0, 0, 0]"/>
-      <sourceAnchor xmi:type="notation:IdentityAnchor" xmi:id="_gCR-s7M3Eeqmquzwsvk0ug"/>
-      <targetAnchor xmi:type="notation:IdentityAnchor" xmi:id="_gCR-tLM3Eeqmquzwsvk0ug"/>
+      <bendpoints xmi:type="notation:RelativeBendpoints" xmi:id="_j3Wgc0_zEeuEueoCkMrBkg" points="[0, 0, 0, 0]$[0, 0, 0, 0]"/>
+      <sourceAnchor xmi:type="notation:IdentityAnchor" xmi:id="_j3WgdE_zEeuEueoCkMrBkg"/>
+      <targetAnchor xmi:type="notation:IdentityAnchor" xmi:id="_j3XHgE_zEeuEueoCkMrBkg"/>
     </edges>
   </notation:Diagram>
 </xmi:XMI>
diff --git a/aiml/org.eclipse.papyrus.aiml.examples/models/kerasmodel.uml b/aiml/org.eclipse.papyrus.aiml.examples/models/kerasmodel.uml
index 77b7b85..c28b04e 100644
--- a/aiml/org.eclipse.papyrus.aiml.examples/models/kerasmodel.uml
+++ b/aiml/org.eclipse.papyrus.aiml.examples/models/kerasmodel.uml
@@ -122,7 +122,7 @@
     </profileApplication>
   </uml:Model>
   <Codegen:GeneratorHint xmi:id="_ggzKkJ6aEeqMUqJDUIBTbw" base_Element="_sfXSgJwiEeqtcKkMgfhA1A" language="_Q3RmYLMyEeqmquzwsvk0ug" generatorID="Keras"/>
-  <AIML.Module:Default_Container xmi:id="_2pbZcLMxEeqmquzwsvk0ug" base_Class="_wkcRgJwiEeqtcKkMgfhA1A"/>
   <Codegen:Language xmi:id="_Q3RmYLMyEeqmquzwsvk0ug" base_Class="_KvaJkJwjEeqtcKkMgfhA1A"/>
   <AIML.Convolution_layers:Conv1d xmi:id="_t2TIMLM3Eeqmquzwsvk0ug" base_Class="_iaXQYLM3Eeqmquzwsvk0ug"/>
+  <AIML.Module:Model xmi:id="_j250gE_zEeuEueoCkMrBkg" base_Class="_wkcRgJwiEeqtcKkMgfhA1A"/>
 </xmi:XMI>
diff --git a/aiml/org.eclipse.papyrus.aiml.gen.keras/META-INF/MANIFEST.MF b/aiml/org.eclipse.papyrus.aiml.gen.keras/META-INF/MANIFEST.MF
index f47ce9f..fa9837e 100644
--- a/aiml/org.eclipse.papyrus.aiml.gen.keras/META-INF/MANIFEST.MF
+++ b/aiml/org.eclipse.papyrus.aiml.gen.keras/META-INF/MANIFEST.MF
@@ -9,7 +9,13 @@
  org.eclipse.xtend.lib;bundle-version="2.2.0",
  org.eclipse.papyrus.designer.languages.common.base;bundle-version="1.1.0",
  org.eclipse.papyrus.designer.languages.common.extensionpoints;bundle-version="1.1.0",
- org.eclipse.papyrus.designer.languages.common.profile;bundle-version="1.1.0"
+ org.eclipse.papyrus.designer.languages.common.profile;bundle-version="1.1.0",
+ org.eclipse.core.resources;bundle-version="3.0.0",
+ org.eclipse.papyrus.uml.tools.utils;bundle-version="3.0.0",
+ org.eclipse.swt;bundle-version="3.0.0",
+ org.eclipse.jface;bundle-version="3.0.0",
+ org.eclipse.papyrus.infra.tools;bundle-version="3.0.0",
+ org.eclipse.papyrus.infra.core.log;bundle-version="1.0.0"
 Export-Package: org.eclipse.papyrus.aiml.gen.keras
 Bundle-Vendor: %providerName
 Bundle-ActivationPolicy: lazy