diff --git a/QMLS Code/Task 2.py b/QMLS Code/Task 2.py
index cba9ca2..c92c55f 100644
--- a/QMLS Code/Task 2.py	
+++ b/QMLS Code/Task 2.py	
@@ -1,4 +1,4 @@
-import torch
+import torch 
 import torch.nn as nn
 import torch.optim as optim
 import numpy as np