diff --git a/Dockerfile b/Dockerfile
index ee487a5..f49f16b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -4,51 +4,60 @@
 ############################################################
 
 #Set base image to Ubuntu
-FROM selenium/standalone-chrome
+FROM jupyter/scipy-notebook
 
-#Update repositor source list
-RUN sudo apt-get update
 
 ################## BEGIN INSTALLATION ######################
 #Install python basics
-RUN sudo apt-get -y install \
-	build-essential \
-	python-dev \
-	python-setuptools
 
 # Create Timezon Variable
 ENV DEBIAN_FRONTEND=noninteractive
 ENV TZ=Asia/Beijing
-# Install tzdata
-RUN sudo apt-get install -y tzdata
 
-# Install python pip
-RUN sudo apt-get -y install python3-pip
+USER root  
+RUN wget -N http://chromedriver.storage.googleapis.com/2.32/chromedriver_linux64.zip
+RUN unzip chromedriver_linux64.zip -d /app
 
-RUN sudo apt-get -y install git-all
+# needed for installing Chrome
+RUN apt-get update && apt-get install -y gnupg2
+RUN apt-get install -y xvfb
 
-#Install other dependancies
-RUN sudo pip3 install matplotlib>=3.2.2
-RUN sudo pip3 install scikit-learn>=0.23.2
-RUN sudo pip3 install scipy==1.5.3
-RUN sudo pip3 install numpy>=1.19.4
-RUN sudo pip3 install pandas>=1.1.3
-RUN sudo pip3 install matplotlib>=3.3.3
-RUN sudo pip3 install requests>=2.24.0
-RUN sudo pip3 install graphviz
-RUN sudo pip3 install altair>=4.1.0
-RUN sudo pip3 install jinja2
-RUN sudo pip3 install pip>=20
-RUN sudo pip3 install pandas-profiling>=1.4.3
-RUN sudo pip3 install psutil>=5.7.2
-RUN sudo pip3 install xgboost>=1.*
-RUN sudo pip3 install lightgbm>=3.*
-RUN sudo pip3 install git+git://github.com/mgelbart/plot-classifier.git
-RUN sudo pip3 install altair-saver==0.5.0
-RUN sudo pip3 install docopt==0.6.2
-RUN sudo pip3 install selenium==3.141.0
-RUN sudo pip3 install webdriver-manager==3.2.2
+# pip install required libraries
+RUN pip install psutil>=5.7.2\
+    git+git://github.com/mgelbart/plot-classifier.git\
+    chromedriver-binary-auto
+
+ # install Chrome to use with altair save
+RUN \
+    wget -q -O - https://dl-ssl.google.com/linux/linux_signing_key.pub | apt-key add - && \
+    echo "deb http://dl.google.com/linux/chrome/deb/ stable main" > /etc/apt/sources.list.d/google.list && \
+    apt-get update && \
+    apt-get install -y google-chrome-stable && \
+    rm -rf /var/lib/apt/lists/*   
 
+#Install other dependancies
+RUN pip3 install matplotlib>=3.2.2
+RUN pip3 install scikit-learn>=0.23.2
+RUN pip3 install scipy==1.5.3
+RUN pip3 install numpy>=1.19.4
+RUN pip3 install pandas>=1.1.3
+RUN pip3 install matplotlib>=3.3.3
+RUN pip3 install requests>=2.24.0
+RUN pip3 install graphviz
+RUN pip3 install altair>=4.1.0
+RUN pip3 install jinja2
+RUN pip3 install pip>=20
+RUN pip3 install pandas-profiling>=1.4.3
+RUN pip3 install psutil>=5.7.2
+RUN pip3 install xgboost>=1.*
+RUN pip3 install lightgbm>=3.*
+RUN pip3 install git+git://github.com/mgelbart/plot-classifier.git
+RUN pip3 install altair-saver==0.5.0
+RUN pip3 install docopt==0.6.2
+RUN pip3 install selenium==3.141.0
+RUN pip3 install webdriver-manager==3.2.2
+
+RUN conda install -y -c anaconda python-chromedriver-binary
 
 
 CMD ["echo Finish Creating Docker Image!"]