diff --git a/UML.dia b/UML.dia
new file mode 100644
index 0000000..3926aef
Binary files /dev/null and b/UML.dia differ
diff --git a/models/User.py b/models/User.py
index 3f67928..353f624 100644
--- a/models/User.py
+++ b/models/User.py
@@ -3,7 +3,7 @@ from db.SQLiteClient import create_user, get_user, get_user_by_email
 
 
 class User(UserMixin):
-    def __init__(self, id, name, email, password=None):
+    def __init__(self, id: int, name: str, email: str, password: str | None = None):
         self.id = id
         self.name = name
         self.email = email