diff --git a/build/build_images.sh b/build/build_images.sh
index 6a2759b64228ad8e4c861a49e326eddb6c50091b..146a4ce476c6c18610c92ada596003745491d230 100755
--- a/build/build_images.sh
+++ b/build/build_images.sh
@@ -2,7 +2,7 @@
 set -e
 
 # Set environment variables
-export VERSION="2.3"
+export VERSION="2.4"
 export DOCKERHUB_REPO="alexdecb"
 
 # Function to build image
diff --git a/build/controller/Dockerfile b/build/controller/Dockerfile
index 48b59f765a24b65070a3a566c11632f9549c7990..35c7e2041dfba7ace74df34488e924d6c3ad7f19 100644
--- a/build/controller/Dockerfile
+++ b/build/controller/Dockerfile
@@ -1,9 +1,12 @@
 FROM onosproject/onos:2.7-latest
 
-COPY ./src/controller ./
 
 RUN apt-get update && \
-    apt-get install wget && \
-    chmod +x ./setup_controller.sh
+    apt-get install -y wget ssh sshpass
+    
+COPY ./src/controller ./
+
+RUN chmod +x ./setup_controller.sh && \
+    chmod +x ./onos_critique.sh
 
 ENTRYPOINT ["./setup_controller.sh"]
\ No newline at end of file