diff --git a/MLProject b/MLProject
index 322d635..34a684f 100644
--- a/MLProject
+++ b/MLProject
@@ -1,15 +1,14 @@
 name: HousePriceModel
 
-docker_env:
-  image: paweldopierala/ium:2.0.0
+conda_env: environment.yml
 
 entry_points:
   main:
     parameters:
       epochs: {type: int, default: 20}
       learning_rate: {type: float, default: 0.001}
-      batch_size: {type: int, default: 20}
-    command: "python train.py {epochs} {learning_rate} {batch_size}"
+      batch_size: {type: int, default: 32}
+    command: "python create_model.py {epochs} {learning_rate} {batch_size}"
 
   test:
-    command: "python test.py"
\ No newline at end of file
+    command: "python evaluate.py"
\ No newline at end of file
diff --git a/environment.yml b/environment.yml
new file mode 100644
index 0000000..839d6c7
--- /dev/null
+++ b/environment.yml
@@ -0,0 +1,13 @@
+name: hp_model_env
+channels:
+  - defaults
+dependencies:
+  - python
+  - pandas
+  - numpy
+  - scikit-learn
+  - matplotlib
+  - tensorflow
+  - keras
+  - mlflow
+  - pip
\ No newline at end of file