diff --git a/README.md b/README.md
index 2472d39..c10c5b4 100644
--- a/README.md
+++ b/README.md
@@ -6,7 +6,16 @@ This is a frontend for the collection of tools used by the zebra.
 ### Contents
 | File | Description |
 |---|---|
-|`wiondow.py`  | The main window. Run from here.|
+| `oneDimView.py` | Widget containing the 1D Mode.|
+| `plot_canvas.py` | Example of a plot. Used by `window.py`. To reuse.|
+| `prepareView.py` | Widget containing the Prepare Mode.|
+| `tab.py` | Example of a tab. Standalone.|
+| `test_a_widget.py` | Window wrapper for easy testing of Widgets.|
+| `tools.py` | Tools used by Widgets.|
+| `twoDimView.py` | Widget containing the 2D Mode.|
+| `welcomeView.py` | Widget containing the Welcome Screen.|
+| `window.py` | Example of a button based interface. Standalone. Legacy. |
+| `zebra.py` | Main Window of the application. Run from here.
 
 <br>
 
diff --git a/test_a_widget.py b/test_a_widget.py
index c324ee4..f0e0bec 100644
--- a/test_a_widget.py
+++ b/test_a_widget.py
@@ -2,8 +2,7 @@ import sys
 
 from PyQt5.QtWidgets import QMainWindow, QApplication, QPushButton,QLabel, QVBoxLayout, QWidget
 
-from two_d_mode import TwoDMode
-from one_d_mode import OneDMode, OneDTable
+
 
 
 
@@ -17,7 +16,7 @@ class TestAWidget(QMainWindow):
         super().__init__()
 
 
-        self.testWidget = OneDMode()
+        self.testWidget = QPushButton()