diff --git a/QMLS Code/Task 1.py b/QMLS Code/Task 1.py
index 35723e4..401138b 100644
--- a/QMLS Code/Task 1.py	
+++ b/QMLS Code/Task 1.py	
@@ -1,6 +1,6 @@
 import torch    
 import torch.nn as nn      
-import torch.optim as optim  
+import torch.optim as optim   
 import torchvision
 import torchvision.transforms as transforms
 import numpy as np