diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index e316f1593621eef4582e536367b4da1689caa357..572e22e92594e04f9af09e7fafb5e0ed90664d82 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -9,6 +9,7 @@ build-and-push-job:
     changes:
       - "**/*.yaml"
       - "README.md"
+      - "screenshots/*.PNG"
   script:
     - echo "{\"auths\":{\"$CI_REGISTRY\":{\"username\":\"$CI_REGISTRY_USER\",\"password\":\"$CI_REGISTRY_PASSWORD\"}}}" > /kaniko/.docker/config.json
     - /kaniko/executor --context $CI_PROJECT_DIR --dockerfile $CI_PROJECT_DIR/Dockerfile --destination $CI_REGISTRY_IMAGE:${CI_COMMIT_SHA:0:8} --destination $CI_REGISTRY_IMAGE:latest
\ No newline at end of file
diff --git a/Dockerfile b/Dockerfile
index 7da4114c1e008e29c00fb8803ac9f14da61418ec..6b4a0e6aa8db4f257dea8b66d79bbb12545635c8 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -40,8 +40,8 @@ RUN useradd -m -s /bin/bash -N -u $NB_UID $NB_USER && \
     chown $NB_USER $CONDA_DIR -R && \
     chown $NB_USER /userdata/kerasData -R && \
     mkdir -p / && \
-    chown $NB_USER / && \
-    chpasswd $NB_USER:ar-noc
+    chpasswd $NB_USER:ar-noc && \
+    usermod -aG sudo $NB_USER
 
 USER $NB_USER
 
diff --git a/README.md b/README.md
index b6d1246e00897d0e658ab53855ab173cdb00fb9e..37a0ecd3e43e5ed305cba6f308b2d001620041ed 100644
--- a/README.md
+++ b/README.md
@@ -127,7 +127,12 @@ The following must be set
 - LRFIND_PLOT_PATH = Where the learning rate finder graph should go
 - TRAINING_PLOT_PATH = Where the training plot graph (loss & accuracy graphs) shoud go 
 3. Loading Data
-There shouldn't be a need to edit this, unless another data loading solution is desired. This section also splits the data into training  
+- Use the load_dataset() function of the load_data notebook ONLY for HPWREN HWB Database 
+- Otherwise, change the script so that the following numpy arrays are generated, if you want to use the prebuild loader in the Training notebook 
+MAKE SURE THESE in load_data.ipynb ![](screenshots/loading_in.PNG)<br /> MATCH THESE in ClassificationExample.ipynb <br /> ![](screenshot/THEOTHEREND.PNG)
+- Use the def loadData(pathToFiles) in ClassificationExample.ipynb
+
+
 4. Image Load Tester
 Tests the images to see if the loading worked
 5. Model Initialization <br />
diff --git a/screenshots/loading_to.PNG b/screenshots/loading_to.PNG
deleted file mode 100755
index 460ec0685c1b81d44e9a16f5f2fc5441a58d6388..0000000000000000000000000000000000000000
Binary files a/screenshots/loading_to.PNG and /dev/null differ