diff --git a/installator.sh b/installator.sh
index 8ecfb1e..7adad00 100644
--- a/installator.sh
+++ b/installator.sh
@@ -4,4 +4,15 @@ sudo apt-get install python-pip python-dev python-virtualenv
 targetDirectory=$1
 virtualenv --system-site-packages "./$targetDirectory"
 source ./$targetDirectory/bin/activate
-sudo -H pip install --upgrade tensorflow pip
+echo "Choose Python version:"
+echo "1 Python 2.7"
+echo "2 Python 3.n"
+read pythonVersion
+case $pythonVersion in
+1)
+  sudo -H pip install --upgrade tensorflow pip
+  ;;
+2) 
+  sudo -H pip3 install --upgrade tensorflow pip
+  ;;
+esac