diff --git a/QMLS Code/Task 1.py b/QMLS Code/Task 1.py
index 03daf90..b3d2613 100644
--- a/QMLS Code/Task 1.py	
+++ b/QMLS Code/Task 1.py	
@@ -2,7 +2,7 @@
 import torch.nn as nn       
 import torch.optim as optim    
 import torchvision
-import torchvision.transforms as transforms
+import torchvision.transforms as transforms 
 import numpy as np
 import os
 # Define data transforms