diff --git a/QMLS Code/Conv CNN.py b/QMLS Code/Conv CNN.py
index 76a772f..9e784ae 100644
--- a/QMLS Code/Conv CNN.py	
+++ b/QMLS Code/Conv CNN.py	
@@ -1,5 +1,5 @@
 import torch 
-import torch.nn as nn 
+import torch.nn as nn
 import torch.nn.functional as F
 class CNN(nn.Module):
     def __init__(self, n):