diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index b0fb431..5e3afbf 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -86,4 +86,4 @@ jobs:
     - uses: actions/checkout@v3
     - name: Build
       working-directory: Kotlin
-      run: make
+      run: make && make run
diff --git a/Kotlin/src/Main.kt b/Kotlin/src/Main.kt
index 330a097..b62b0ef 100644
--- a/Kotlin/src/Main.kt
+++ b/Kotlin/src/Main.kt
@@ -47,10 +47,9 @@ fun main() {
 
   val lr = 1.0
   val ITERS = 4000
-  for (i in 1..ITERS) {
-    for (it in trainingData) {
-      trainer.train(it.first, it.second, lr)
-    }
+  for (i in 0..<ITERS) {
+    val it = trainingData[i % 4]
+    trainer.train(it.first, it.second, lr)
   }
 
   println("Result after ${ITERS} iterations")
diff --git a/README.md b/README.md
index 6f77a47..ffbc7ec 100644
--- a/README.md
+++ b/README.md
@@ -218,8 +218,6 @@ This is the current status of the implementations available. We follow a maturit
 
 > Note! The Python implementation is only here as a reference. If you are using Python you already
 > have access to all ai tools and libraries you need.
->
-> Note 2! Kotlin seems to be off, not sure why yet.
 
 #### 4.3.1. Sample Output