diff --git a/07_final_assignment/baboonSimulation.R b/07_final_assignment/baboonSimulation.R
index 9959aee..05b480d 100644
--- a/07_final_assignment/baboonSimulation.R
+++ b/07_final_assignment/baboonSimulation.R
@@ -198,7 +198,7 @@
     }
 
     list(
-        Weights=give_weights,
+        GetWeights=give_weights,
         Learner=learner,
         Rate=rate
     )
@@ -208,15 +208,12 @@
     cueset <- c("a","b","c")
 
     monkey <- MakeMonkey(cueset)
-    monkey$Learner("a_b","Word")
-    monkey$Learner("b_c","Nonword")
-    monkey$Learner("a_b","Word")
-    monkey$Learner("b_c","Nonword")
-    monkey$Learner("a_b","Word")
-    monkey$Learner("b_c","Nonword")
-    monkey$Learner("a_b","Word")
-    monkey$Learner("b_c","Nonword")
-
+    for (i in 1:1000){
+        monkey$Learner("a_b","Word")
+        monkey$Learner("b_c","Nonword")
+    }
+ 
+    print(monkey$GetWeights())
     expect_equal(monkey$Rate("a"), "Word")
     expect_equal(monkey$Rate("c"), "Nonword")
  })