diff --git a/docker/Dockerfile b/docker/Dockerfile
index decaadc..b50e53b 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -68,5 +68,12 @@ RUN chmod a+x /opt/entrypoint.sh && \
 ADD ./spark-azure-nytaxi-1.0-SNAPSHOT-jar-with-dependencies.jar /root/spark-azure-nytaxi-1.0-SNAPSHOT.jar
 RUN cp /root/spark-azure-nytaxi-1.0-SNAPSHOT.jar /opt/bigdl-${BIGDL_VERSION}/jars/
 ADD ./run_azure_nytaxi.sh /root/run_azure_nytaxi.sh
+ADD ./run_spark_pi.sh /root/run_spark_pi.sh
+ADD ./run_nytaxi_k8s.sh /root/run_nytaxi_k8s.sh
+ADD ./driver.yaml /root/driver.yaml
+ADD ./executor.yaml /root/executor.yaml
+
+# install azure cli
+RUN curl -sL https://aka.ms/InstallAzureCLIDeb | bash
 
 ENTRYPOINT [ "/opt/entrypoint.sh" ]