diff --git a/QMLS Code/Task 2.py b/QMLS Code/Task 2.py
index c92c55f..cba9ca2 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