diff --git a/rgbAI/lib/func.py b/rgbAI/lib/func.py
index 5cf23ed..210f904 100644
--- a/rgbAI/lib/func.py
+++ b/rgbAI/lib/func.py
@@ -38,14 +38,14 @@ class AIlib:
 
     def propDer( dCost, dProp ):
         # Calculate the partial derivative for that prop
-        print("################")
-        print(dCost, dProp)
+        #print("################")
+        #print(dCost, dProp)
         return dCost / dProp
 
     def gradient( inp:np.array, obj, theta:float, maxLayer:int, layerIndex: int=0, grads=None, obj1=None, obj2=None ): # Calculate the gradient for that prop
         # Check if grads exists, if not create the buffer
         if( not grads ):
-            grads = [None] * maxLayer
+            grads = [None] * (maxLayer+1)
 
         # Create new instances of the object
         if( not obj1 or not obj2 ):
diff --git a/rgbAI/main.py b/rgbAI/main.py
index d02231a..1adc53a 100755
--- a/rgbAI/main.py
+++ b/rgbAI/main.py
@@ -55,7 +55,7 @@ class rgb(object):
 def init():
     bot = rgb()
 
-    bot.traintimes = 100
+    bot.traintimes = 1
     bot.train()