diff --git a/QMLS Code/Task 1.py b/QMLS Code/Task 1.py
index 1ada488..948cc70 100644
--- a/QMLS Code/Task 1.py	
+++ b/QMLS Code/Task 1.py	
@@ -1,5 +1,5 @@
 import torch  
-import torch.nn as nn   
+import torch.nn as nn    
 import torch.optim as optim
 import torchvision
 import torchvision.transforms as transforms