diff --git a/Dockerfile b/Dockerfile
index c54e93b..a48715f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -4,10 +4,20 @@ ENV DEBIAN_FRONTEND=noninteractive
 
 LABEL com.amazonaws.sagemaker.capabilities.accept-bind-to-port=true
 
-RUN apt-get update && apt-get -y install apt-utils net-tools apt-transport-https wget curl nginx git maven
+RUN apt-get update \
+ && apt-get -y install apt-utils \
+    net-tools \
+    apt-transport-https \
+    wget \
+    curl \
+    nginx \
+    git \
+    maven \
+    make \
+    gcc \
+    zlib1g-dev
 
 RUN apt -y update
-RUN apt -y install python3.6
 
 ARG OPENSSL_VERSION=1.1.1g
 ARG PYTHON=python3
@@ -36,11 +46,7 @@ RUN wget -q https://www.python.org/ftp/python/$PYTHON_VERSION/Python-$PYTHON_VER
  && ${PIP} --no-cache-dir install --upgrade pip
 
 # Remove other Python installations.
-RUN apt -y purge --auto-remove libpython2.7 \
- && apt -y purge --auto-remove libpython3.7 \
- && apt -y purge --auto-remove python3.7 \
- && apt -y purge --auto-remove python2.7 \
- && apt-get clean \
+RUN apt-get clean \
  && rm -rf /var/lib/apt/lists/*
 
 COPY / /sagemaker-sparkml-model-server
diff --git a/pom.xml b/pom.xml
index ea84d1a..2c98307 100644
--- a/pom.xml
+++ b/pom.xml
@@ -204,5 +204,6 @@
   </dependencies>
   <properties>
     <java.version>1.8</java.version>
+    <log4j2.version>2.17.1</log4j2.version>
   </properties>
 </project>