diff --git a/src/main/java/de/th_luebeck/ws25/LabeledSequence.java b/src/main/java/de/th_luebeck/ws25/LabeledSequence.java
index d7b742e..2a4d982 100644
--- a/src/main/java/de/th_luebeck/ws25/LabeledSequence.java
+++ b/src/main/java/de/th_luebeck/ws25/LabeledSequence.java
@@ -1,4 +1,6 @@
 package de.th_luebeck.ws25;
 
-public record LabeledSequence(Label label, int[] trainData) {
+import java.util.List;
+
+public record LabeledSequence(Label label, List<Integer> sequence) {
 }