diff --git a/architectures/Model.py b/architectures/Model.py
index 50d506d..256b050 100644
--- a/architectures/Model.py
+++ b/architectures/Model.py
@@ -16,7 +16,6 @@ class Model(ABC):
 
     @abstractmethod
     def get(self):
-        # return the model
         pass
 
     def train(self, epochs, loader, rate):
@@ -66,7 +65,7 @@ class Model(ABC):
         save_dir = Path("trained_models")
         save_dir.mkdir(parents=True, exist_ok=True)
 
-        # 2. Determine filename (Default to class name if not provided)
+        # Determine filename (Default to class name if not provided)
         if filename is None:
             filename = f"{self.__class__.__name__.lower()}.pth"