diff --git a/environment_setup/docker-image-pipeline.yml b/environment_setup/docker-image-pipeline.yml
index 633caf1f..7678d96d 100644
--- a/environment_setup/docker-image-pipeline.yml
+++ b/environment_setup/docker-image-pipeline.yml
@@ -1,8 +1,8 @@
 resources:
 - repo: self
 
-queue:
-  name: Hosted Ubuntu 1604
+pool: 
+  vmImage: 'ubuntu-latest'
 
 trigger:
   branches:
@@ -24,6 +24,8 @@ steps:
       command: buildAndPush
       containerRegistry: '$(containerRegistry)'
       repository: '$(imageName)'
-      tags: 'latest'
+      tags: |
+       $(Build.BuildNumber)
+       latest
       buildContext: '$(Build.SourcesDirectory)' 
       dockerFile: '$(Build.SourcesDirectory)/environment_setup/Dockerfile'
diff --git a/environment_setup/requirements.txt b/environment_setup/requirements.txt
index 8a086c4d..e7d7cc7b 100644
--- a/environment_setup/requirements.txt
+++ b/environment_setup/requirements.txt
@@ -1,8 +1,9 @@
-pytest==4.3.0
+pytest>=5.3
 requests>=2.22
-azureml>=0.2
+numpy>=1.17
+pandas>=0.25
 azureml-sdk>=1.0
 python-dotenv>=0.10.3
-flake8
+flake8>=3.7
 flake8_formatter_junit_xml
-azure-cli==2.0.71
\ No newline at end of file
+azure-cli==2.0.76