diff --git a/function.py b/function.py
index b19b944..117e1a7 100644
--- a/function.py
+++ b/function.py
@@ -117,8 +117,10 @@ def load_data(filepath):
                 values = [float(v) for v in values]
                 if values[22] == 2 or values[22] == 3:
                     labels.append(0)
+                    data.append(values[:2] + values[3:23])
                 elif values[22] == 1:
                     labels.append(1)
+                    data.append(values[:2] + values[3:23])
                 else:
                     continue