diff --git a/unet/LaunchBatchPrediction.py b/unet/LaunchBatchPrediction.py
index 549dd71e13978bcc261893e088bfd63bd4da9998..6ee7b04b85d9c38afaf20eb8b799433e030b4d56 100644
--- a/unet/LaunchBatchPrediction.py
+++ b/unet/LaunchBatchPrediction.py
@@ -3,69 +3,51 @@
 Created on Tue Nov 19 17:38:58 2019
 """
 
-from PyQt5.QtWidgets import QApplication, QMainWindow, QMenu, QVBoxLayout, QSizePolicy, QMessageBox, QWidget, QPushButton, QShortcut, QComboBox, QDialog, QDialogButtonBox, QInputDialog, QLineEdit, QFormLayout, QLabel, QListWidget, QAbstractItemView, QCheckBox
+from PyQt5.QtWidgets import (QDialog, QDialogButtonBox, QLineEdit, QFormLayout, 
+                             QLabel, QListWidget, QAbstractItemView, QCheckBox)
 from PyQt5 import QtGui
-#from PyQt5.QtGui import QIcon, QKeySequence
-from PyQt5.QtCore import pyqtSignal, QObject, Qt
-#import PyQt package, allows for GUI interactions
-import pdb
-class CustomDialog(QDialog):
 
+
+class CustomDialog(QDialog):
     def __init__(self, *args, **kwargs):
-            
         super(CustomDialog, self).__init__(*args, **kwargs)
         
         app, = args
         maxtimeindex = app.reader.sizet
-        maxfovindex = app.reader.Npos
         
         self.setWindowTitle("Launch NN")
         self.setGeometry(100,100, 500,200)
         
-
-        
         self.entry1 = QLineEdit()
         self.entry1.setValidator(QtGui.QIntValidator(0,int(maxtimeindex-1)))
-
-        
         self.entry2 = QLineEdit()
         self.entry2.setValidator(QtGui.QIntValidator(0,int(maxtimeindex-1)))
-
         
+        # FOV dialog
         self.listfov = QListWidget()
         self.listfov.setSelectionMode(QAbstractItemView.MultiSelection)
-        
         for f in range(0, app.reader.Npos):
             self.listfov.addItem('Field of View {}'.format(f+1))
 
-        
         self.labeltime = QLabel("Enter range ({}-{}) for time axis".format(0, app.reader.sizet-1))
         
         self.entry_threshold = QLineEdit()
         self.entry_threshold.setValidator(QtGui.QDoubleValidator())
         self.entry_threshold.setText('0.5')
         
-        
-        
         self.entry_segmentation = QLineEdit()
         self.entry_segmentation.setValidator(QtGui.QIntValidator())
         self.entry_segmentation.setText('10')
         
         self.tracking_checkbox = QCheckBox()
         
-        
-        
         flo = QFormLayout()
         flo.addWidget(self.labeltime)
         flo.addRow('Lower Boundary for time axis', self.entry1)
-        flo.addRow('Upper Boundary for time axis', self.entry2)
-        
-        
+        flo.addRow('Upper Boundary for time axis', self.entry2)        
         flo.addRow('Select Field(s) of fiew from  the list', self.listfov)
-        
         flo.addRow('Enter a threshold value', self.entry_threshold)
         flo.addRow('Enter a segmentation value', self.entry_segmentation)
-        
         flo.addRow('Apply Cell Tracker', self.tracking_checkbox)
         
         QBtn = QDialogButtonBox.Ok | QDialogButtonBox.Cancel