diff --git a/Dockerfile b/Dockerfile
index f20a323..dc55450 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,24 +1,32 @@
 FROM quay.io/jupyter/minimal-notebook:afe30f0c9ad8
 
+# Copy the conda lock file into the container
 COPY conda-linux-64.lock /tmp/conda-linux-64.lock
 
+# Switch to root user to install additional packages
 USER root
 
-# install lmodern for Quarto PDF rendering
-RUN sudo apt update \
-        && sudo apt install -y \
-        lmodern
+# Install lmodern and make for Quarto PDF rendering and build tools
+RUN apt update && apt install -y lmodern make \
+    && apt clean \
+    && rm -rf /var/lib/apt/lists/*
 
+# Fix permissions
+RUN fix-permissions "/home/${NB_USER}"
+
+# Switch back to the notebook user
 USER $NB_UID
 
+# Update mamba environment and clean up
 RUN mamba update --quiet --file /tmp/conda-linux-64.lock \
     && mamba clean --all -y -f \
-    && fix-permissions "${CONDA_DIR}" \
-    && fix-permissions "/home/${NB_USER}"
-    
-RUN pip install \
+    && fix-permissions "${CONDA_DIR}"
+
+# Install Python packages with pip
+RUN pip install --no-cache-dir \
     deepchecks==0.18.1 \
     seaborn==0.13.2 \
     altair-ally==0.1.1
 
-RUN echo "Done Building Container!!"
\ No newline at end of file
+# Log container build completion
+RUN echo "Done Building Container!!"
diff --git a/docker-compose.yml b/docker-compose.yml
index a05a142..862053a 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,7 +1,7 @@
 services:
   # run jupyter notebook inside jupyter 
   jupyter-notebook:
-    image:  piloxita/dsci-522-2425-team35-heart_disease_diagnostic_machine:1c53335
+    image:  piloxita/dsci-522-2425-team35-heart_disease_diagnostic_machine:d065521
     ports:
       - "9999:8888"
     volumes: