diff --git a/lab7/simple_regression_lab7.py b/lab7/simple_regression_lab7.py
index 33c6e4f..c12eeec 100644
--- a/lab7/simple_regression_lab7.py
+++ b/lab7/simple_regression_lab7.py
@@ -81,9 +81,10 @@ def train(units, learning_rate, _run):
     # Przykładowo zwracamy coś w charakterze wyników, żeby było widoczne w plikach zapisanych przez obserwator
     hist = pd.DataFrame(history.history)
     hist['epoch'] = history.epoch
-    _run.info["train_results"] = str(hist.tail())
-    return hist.tail()
+    _run.log_scalar('final.training.loss', hist['loss'].iloc[-1])
+    print(hist['loss'].iloc[-1])
+    print(type(hist['loss'].iloc[-1]))
 
 @ex.automain
 def main(units, learning_rate):
-    print(train())
\ No newline at end of file
+    train()
\ No newline at end of file