diff --git a/src/wireviz/DataClasses.py b/src/wireviz/DataClasses.py
index a1df6ff..6ffc044 100644
--- a/src/wireviz/DataClasses.py
+++ b/src/wireviz/DataClasses.py
@@ -9,12 +9,16 @@ from wireviz import wv_colors
@dataclass
class Image:
+ # Attributes of the image object :
src: str
- scale: str = "false"
+ scale: Optional[str] = "false" # false | true | width | height | both
+ # Attributes of the image cell