diff --git a/unit-of-work/etc/unit-of-work.urm.png b/unit-of-work/etc/unit-of-work.urm.png
index 3a7c0d377..bb192af54 100644
Binary files a/unit-of-work/etc/unit-of-work.urm.png and b/unit-of-work/etc/unit-of-work.urm.png differ
diff --git a/unit-of-work/etc/unit-of-work.urm.puml b/unit-of-work/etc/unit-of-work.urm.puml
index 7f7920daf..7767a2e32 100644
--- a/unit-of-work/etc/unit-of-work.urm.puml
+++ b/unit-of-work/etc/unit-of-work.urm.puml
@@ -1,5 +1,9 @@
 @startuml
 package com.iluwatar.unitofwork {
+  class App {
+    + App()
+    + main(args : String[]) {static}
+  }
   interface IUnitOfWork<T> {
     + DELETE : String {static}
     + INSERT : String {static}
@@ -24,10 +28,6 @@ package com.iluwatar.unitofwork {
     + insert(student : Student)
     + modify(student : Student)
   }
-  class StudentManagementApp {
-    + StudentManagementApp()
-    + main(args : String[]) {static}
-  }
   class StudentRepository {
     - LOGGER : Logger {static}
     - context : Map<String, List<Student>>