diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/MLmodel b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/MLmodel
new file mode 100644
index 0000000..76256a0
--- /dev/null
+++ b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/MLmodel
@@ -0,0 +1,14 @@
+artifact_path: vgsales_model.h5
+flavors:
+  keras:
+    data: data
+    keras_module: tensorflow.keras
+    keras_version: 2.4.0
+    save_format: tf
+  python_function:
+    data: data
+    env: conda.yaml
+    loader_module: mlflow.keras
+    python_version: 3.8.5
+run_id: 24e424dedaad4645b9d7bd5b0dd826ca
+utc_time_created: '2021-05-17 10:38:48.722780'
diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/conda.yaml b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/conda.yaml
new file mode 100644
index 0000000..3fa4018
--- /dev/null
+++ b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/conda.yaml
@@ -0,0 +1,10 @@
+channels:
+- defaults
+- conda-forge
+dependencies:
+- python=3.8.5
+- pip
+- pip:
+  - mlflow
+  - tensorflow==2.4.1
+name: mlflow-env
diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/data/keras_module.txt b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/data/keras_module.txt
new file mode 100644
index 0000000..2c73dfd
--- /dev/null
+++ b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/data/keras_module.txt
@@ -0,0 +1 @@
+tensorflow.keras
\ No newline at end of file
diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/data/model/saved_model.pb b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/data/model/saved_model.pb
new file mode 100644
index 0000000..bd3ca01
Binary files /dev/null and b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/data/model/saved_model.pb differ
diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/data/model/variables/variables.data-00000-of-00001 b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/data/model/variables/variables.data-00000-of-00001
new file mode 100644
index 0000000..baaa6ce
Binary files /dev/null and b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/data/model/variables/variables.data-00000-of-00001 differ
diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/data/model/variables/variables.index b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/data/model/variables/variables.index
new file mode 100644
index 0000000..cdfe2a8
Binary files /dev/null and b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/data/model/variables/variables.index differ
diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/data/save_format.txt b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/data/save_format.txt
new file mode 100644
index 0000000..439ab29
--- /dev/null
+++ b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts/vgsales_model.h5/data/save_format.txt
@@ -0,0 +1 @@
+tf
\ No newline at end of file
diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/meta.yaml b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/meta.yaml
new file mode 100644
index 0000000..ba07d80
--- /dev/null
+++ b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/meta.yaml
@@ -0,0 +1,15 @@
+artifact_uri: file:///home/tomasz/IUM/ium_434695/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/artifacts
+end_time: 1621247929620
+entry_point_name: ''
+experiment_id: '0'
+lifecycle_stage: active
+name: ''
+run_id: 24e424dedaad4645b9d7bd5b0dd826ca
+run_uuid: 24e424dedaad4645b9d7bd5b0dd826ca
+source_name: ''
+source_type: 4
+source_version: ''
+start_time: 1621247902981
+status: 3
+tags: []
+user_id: tomasz
diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/metrics/rmse b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/metrics/rmse
new file mode 100644
index 0000000..662e88b
--- /dev/null
+++ b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/metrics/rmse
@@ -0,0 +1 @@
+1621247928720 0.032181188679670125 0
diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/params/batch_size b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/params/batch_size
new file mode 100644
index 0000000..19c7bdb
--- /dev/null
+++ b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/params/batch_size
@@ -0,0 +1 @@
+16
\ No newline at end of file
diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/params/epochs b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/params/epochs
new file mode 100644
index 0000000..3f10ffe
--- /dev/null
+++ b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/params/epochs
@@ -0,0 +1 @@
+15
\ No newline at end of file
diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/tags/mlflow.log-model.history b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/tags/mlflow.log-model.history
new file mode 100644
index 0000000..c825cdc
--- /dev/null
+++ b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/tags/mlflow.log-model.history
@@ -0,0 +1 @@
+[{"run_id": "24e424dedaad4645b9d7bd5b0dd826ca", "artifact_path": "vgsales_model.h5", "utc_time_created": "2021-05-17 10:38:48.722780", "flavors": {"keras": {"keras_module": "tensorflow.keras", "keras_version": "2.4.0", "save_format": "tf", "data": "data"}, "python_function": {"loader_module": "mlflow.keras", "python_version": "3.8.5", "data": "data", "env": "conda.yaml"}}}]
\ No newline at end of file
diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/tags/mlflow.source.git.commit b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/tags/mlflow.source.git.commit
new file mode 100644
index 0000000..d2aa668
--- /dev/null
+++ b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/tags/mlflow.source.git.commit
@@ -0,0 +1 @@
+3725a34755f266ea631bf115a1499767f27fbb01
\ No newline at end of file
diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/tags/mlflow.source.name b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/tags/mlflow.source.name
new file mode 100644
index 0000000..939bfd8
--- /dev/null
+++ b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/tags/mlflow.source.name
@@ -0,0 +1 @@
+/home/tomasz/IUM/ium_434695/vgsales-mlflow.py
\ No newline at end of file
diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/tags/mlflow.source.type b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/tags/mlflow.source.type
new file mode 100644
index 0000000..0c2c1fe
--- /dev/null
+++ b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/tags/mlflow.source.type
@@ -0,0 +1 @@
+LOCAL
\ No newline at end of file
diff --git a/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/tags/mlflow.user b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/tags/mlflow.user
new file mode 100644
index 0000000..4b82f83
--- /dev/null
+++ b/mlruns/0/24e424dedaad4645b9d7bd5b0dd826ca/tags/mlflow.user
@@ -0,0 +1 @@
+tomasz
\ No newline at end of file
diff --git a/mlruns/0/meta.yaml b/mlruns/0/meta.yaml
new file mode 100644
index 0000000..8a02dc3
--- /dev/null
+++ b/mlruns/0/meta.yaml
@@ -0,0 +1,4 @@
+artifact_location: file:///home/tomasz/IUM/ium_434695/mlruns/0
+experiment_id: '0'
+lifecycle_stage: active
+name: Default
diff --git a/vgsales-mlflow.py b/vgsales-mlflow.py
index 219de18..0ee329c 100644
--- a/vgsales-mlflow.py
+++ b/vgsales-mlflow.py
@@ -1,5 +1,5 @@
 import sys
-from keras.backend import batch_dot, mean
+from tensorflow.keras.backend import batch_dot, mean
 import pandas as pd
 import numpy as np
 from six import int2byte
diff --git a/vgsales_model.h5 b/vgsales_model.h5
new file mode 100644
index 0000000..63aca41
Binary files /dev/null and b/vgsales_model.h5 differ