diff --git a/environment.yml b/environment.yml
index 53b5a29..857ea8e 100644
--- a/environment.yml
+++ b/environment.yml
@@ -1,6 +1,5 @@
 name: handson-ml
 dependencies:
-  - python=3.5
   - scikit-learn
   - matplotlib
   - jupyter
@@ -13,3 +12,4 @@ dependencies:
     - graphviz
     - watermark
     - urlextract
+    - xgboost