diff --git a/MainCode/adalm1000_logger.py b/MainCode/adalm1000_logger.py
index b4c875e..68f95b4 100644
--- a/MainCode/adalm1000_logger.py
+++ b/MainCode/adalm1000_logger.py
@@ -11,8 +11,8 @@ from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
 from matplotlib.figure import Figure
 from collections import deque
 
-from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, QLabel, 
-                            QPushButton, QLineEdit, QCheckBox, QFrame, QMessageBox, QFileDialog)
+from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, QGridLayout, QLabel, 
+                             QPushButton, QLineEdit, QCheckBox, QFrame, QMessageBox, QFileDialog)
 from PyQt5.QtCore import Qt, QTimer, pyqtSignal, pyqtSlot, QObject
 import pysmu