diff --git a/Dockerfile b/Dockerfile
index 9604fac..0ab611b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -13,9 +13,7 @@ ENV YARN_RESOURCEMANAGER_USER=root
 
 # Install necessary dependencies
 RUN apt-get update && \
-    apt-get install -y ssh && \
-    apt-get install -y openjdk-8-jdk && \
-    apt-get install -y neovim && \
+    apt-get install -y ssh openjdk-8-jdk neovim junit && \
     rm -rf /var/lib/apt/lists/*
 
 # Download and extract Hadoop
@@ -59,4 +57,4 @@ RUN echo "export HADOOP_CLASSPATH+=\" \$HADOOP_HOME/lib/*.jar\"" >> /usr/local/h
 
 
 # Expose necessary ports
-EXPOSE 9870 8088 9000
\ No newline at end of file
+EXPOSE 9870 8088 9000