diff --git a/architecture.py b/architecture.py
index f503854..45ab53d 100644
--- a/architecture.py
+++ b/architecture.py
@@ -4,8 +4,7 @@ import torch.nn
 
 class MyCNN(torch.nn.Module):
     def __init__(self,
-                 input_channels: int,
-                 input_size: tuple[int, int]):
+                 input_channels: int):
         super().__init__()
 
         self.layers = torch.nn.Sequential(
@@ -51,4 +50,4 @@ class MyCNN(torch.nn.Module):
         return str(self.layers)
 
 
-model = MyCNN(input_channels=1, input_size=(100, 100))
+model = MyCNN(input_channels=1)
diff --git a/submit_attempts/Nr1/architecture.py b/submit_attempts/Nr1/architecture.py
index f503854..45ab53d 100644
--- a/submit_attempts/Nr1/architecture.py
+++ b/submit_attempts/Nr1/architecture.py
@@ -4,8 +4,7 @@ import torch.nn
 
 class MyCNN(torch.nn.Module):
     def __init__(self,
-                 input_channels: int,
-                 input_size: tuple[int, int]):
+                 input_channels: int):
         super().__init__()
 
         self.layers = torch.nn.Sequential(
@@ -51,4 +50,4 @@ class MyCNN(torch.nn.Module):
         return str(self.layers)
 
 
-model = MyCNN(input_channels=1, input_size=(100, 100))
+model = MyCNN(input_channels=1)
diff --git a/submit_attempts/Nr1/model.pt b/submit_attempts/Nr1/model.pth
similarity index 100%
rename from submit_attempts/Nr1/model.pt
rename to submit_attempts/Nr1/model.pth