diff --git a/QMLS Code/Task 1.py b/QMLS Code/Task 1.py
index 401138b..bd06b7e 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