diff --git a/docs/UML/Class Diagram.puml b/docs/UML/Class Diagram.puml
index 67eca8f..ccce635 100644
--- a/docs/UML/Class Diagram.puml	
+++ b/docs/UML/Class Diagram.puml	
@@ -3,50 +3,50 @@
 class GUI{
     This is a singleton
     ---
-    - Image currentImage
-    - ImageManipulation[] imageManipulations
+    - ImageContainer _currentImage
+    - ImageManipulation[] _imageManipulations
     + none initialise()
 }
 
 
-class Image{
-    - imagedata image
+class ImageContainer{
+    - imagedata _imageData
     + none loadImage(path: str)
     + imagedata getImage()
     + pair<int,int> getDimensions()
 }
 
 abstract class ImageManipulation{
-    +none manipulateImage(Image, Parameters)
+    +none manipulateImage(ImageContainer, Parameters)
     +str[] getParameters()
     +str getManipulationName()
 }
 
 class CropImage{
-- str name
+- str _name
 }
 
 class GreyScaleImage{
-- str name
+- str _name
 }
 
 class SmoothImage{
-- str name
+- str _name
 }
 
 class RotateImage{
-- str name
+- str _name
 }
 
 class HSVImage{
-- str
+- str _name
 }
 
 class HueShiftImage{
-- str
+- str _name
 }
 
-GUI ..> Image
+GUI ..> ImageContainer
 GUI ..> ImageManipulation
 
 CropImage  ..|> ImageManipulation