diff --git a/docs/Diagrams/Images/Itemlist/EditCommand_ClassDiagram.png b/docs/Diagrams/Images/Itemlist/EditCommand_ClassDiagram.png
index c688a3582..3c6ccfedb 100644
Binary files a/docs/Diagrams/Images/Itemlist/EditCommand_ClassDiagram.png and b/docs/Diagrams/Images/Itemlist/EditCommand_ClassDiagram.png differ
diff --git a/docs/Diagrams/PUML files/Itemlist/EditCommand_ClassDiagram.puml b/docs/Diagrams/PUML files/Itemlist/EditCommand_ClassDiagram.puml
index d625b4a32..4328695f4 100644
--- a/docs/Diagrams/PUML files/Itemlist/EditCommand_ClassDiagram.puml	
+++ b/docs/Diagrams/PUML files/Itemlist/EditCommand_ClassDiagram.puml	
@@ -1,6 +1,5 @@
 @startuml
 !include ../Style.puml
-
 class EditCommand {
     - itemName : String
     - newItemName : String
@@ -9,15 +8,7 @@ class EditCommand {
     - newCategory : String
     - newBuyPrice : float
     - newSellPrice : float
-
-    + EditCommand(itemName: String, newItemName: String, newQuantity: int, newUnitOfMeasurement: String, newCategory: String, newBuyPrice: float, newSellPrice: float)
-    + getItemName() : String
-    + getNewItemName() : String
-    + getNewQuantity() : int
-    + getNewUnitOfMeasurement() : String
-    + getNewCategory() : String
-    + getNewBuyPrice() : float
-    + getNewSellPrice() : float
+    + EditCommand(...)
     + execute() : void
 }
 
@@ -28,45 +19,37 @@ class Item {
     - category : String
     - buyPrice : float
     - sellPrice : float
-    - isOOS : boolean
-
-    + setItemName(newName: String)
-    + setQuantity(newQuantity: int)
-    + setUnitOfMeasurement(newUnit: String)
-    + setCategory(newCategory: String)
-    + setBuyPrice(newPrice: float)
-    + setSellPrice(newPrice: float)
-    + getItemName() : String
-    + getQuantity() : int
-    + getUnitOfMeasurement() : String
-    + getCategory() : String
-    + getBuyPrice() : float
-    + getSellPrice() : float
-    + markOOS()
-    + unmarkOOS()
 }
 
 class Itemlist {
-    - items : ArrayList<Item>
+    + getItems() : List<Item>
+    + getItem(index: int) : Item'
+}
 
-    + getItems() : ArrayList<Item>
-    + getItem(index: int) : Item
-    + overwriteFile(items : ArrayList<Item>)
+class Promotion {
+    - itemName : String
+    + setItemName(newItemName: String) : void
 }
 
-class TextUi {
-    + replyToUser(message: String)
-    + showEditMessage(oldValue: String, attribute: String, newValue: String)
+class Promotionlist {
+    + itemIsOnPromo(itemName: String) : boolean
+    + getPromotion(itemName: String) : Promotion
+    + getAllPromotion() : List<Promotion>
 }
 
 class Storage {
-    + overwriteFile(items : ArrayList<Item>)
+    + overwriteFile(items: List<Item>) : void
+}
+
+class PromotionStorage {
+    + overwritePromotionFile(promotions: List<Promotion>) : void
 }
 
-EditCommand -- "0..*" Item
-EditCommand -- "1" Itemlist
-EditCommand -- "1" TextUi
-EditCommand -- "1" Storage
-hide circle
-skinparam classAttributeIconSize 0
+EditCommand -> Item
+EditCommand --> Itemlist
+EditCommand --> Promotion : uses >
+EditCommand --> Promotionlist
+Storage <-- EditCommand
+PromotionStorage <-- EditCommand
+
 @enduml