diff --git a/QMLS Code/Task 1.py b/QMLS Code/Task 1.py
index 390bd3d..b9f8167 100644
--- a/QMLS Code/Task 1.py	
+++ b/QMLS Code/Task 1.py	
@@ -1,4 +1,4 @@
-import torch
+import torch 
 import torch.nn as nn
 import torch.optim as optim
 import torchvision