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