diff --git a/lab5/train/train.py b/lab5/train/train.py
index 0f3c519..099c934 100644
--- a/lab5/train/train.py
+++ b/lab5/train/train.py
@@ -16,12 +16,18 @@ Y_test = pd.get_dummies(Y_test)
 Y_valid = X_valid.pop('stabf')
 Y_valid = pd.get_dummies(Y_valid)
 
-model = tensorflow.keras.Sequential([
-                          layers.Input(shape=(12,)),
-                          layers.Dense(32),
-                          layers.Dense(16),
-                          layers.Dense(2, activation='softmax')
-])
+# model = tensorflow.keras.Sequential([
+#                           layers.Input(shape=(12,)),
+#                           layers.Dense(32),
+#                           layers.Dense(16),
+#                           layers.Dense(2, activation='softmax')
+# ])
+
+model = tensorflow.keras.Sequential()
+model.add(layers.Input(shape=(12,)))
+model.add(layers.Dense(32))
+model.add(layers.Dense(16))
+model.add(layers.Dense(2, activation='softmax'))
 
 model.compile(
     loss=tensorflow.keras.losses.BinaryCrossentropy(),