diff --git a/build.gradle b/build.gradle
index 48a4181..e0ab2c9 100644
--- a/build.gradle
+++ b/build.gradle
@@ -1,6 +1,7 @@
 plugins {
     id 'java'
     id 'application'
+    id 'com.github.johnrengelman.shadow' version '5.2.0'
 }
 
 group 'net.lwenstrom.jan'
diff --git a/src/main/java/example/JumpingDino.java b/src/main/java/example/JumpingDino.java
index 88f6c16..c191d9b 100644
--- a/src/main/java/example/JumpingDino.java
+++ b/src/main/java/example/JumpingDino.java
@@ -5,7 +5,6 @@ import core.algo.Method;
 import core.controller.RLController;
 import core.controller.RLControllerGUI;
 import evironment.jumpingDino.DinoAction;
-import evironment.jumpingDino.DinoWorld;
 import evironment.jumpingDino.DinoWorldAdvanced;
 
 public class JumpingDino {
@@ -21,7 +20,7 @@ public class JumpingDino {
         rl.setDiscountFactor(1f);
         rl.setEpsilon(0.15f);
         rl.setLearningRate(1f);
-        rl.setNrOfEpisodes(50000);
+        rl.setNrOfEpisodes(50001);
         rl.start();
     }
 }