diff --git a/QMLS Code/Task 4.py b/QMLS Code/Task 4.py
index bb08744..4a76e4d 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