diff --git a/metrics.py b/metrics.py
index d88377f..cf6852f 100644
--- a/metrics.py
+++ b/metrics.py
@@ -6,12 +6,9 @@ import sys
 data = pd.read_csv('powerlifting_test_predictions.csv')
 y_pred = data['predicted_TotalKg']
 y_test = data['actual_TotalKg']
-y_test_binary = (y_test >= 3).astype(int)
-
-# build_number = sys.argv[1]
-build_number = 1
 
+build_number = sys.argv[1]
 rmse = sqrt(mean_squared_error(y_test, y_pred))
 
 with open(r"metrics.txt", "a") as f:
-    f.write(f"{123},{1}\n")
\ No newline at end of file
+    f.write(f"{build_number},{rmse}\n")
\ No newline at end of file