diff --git a/netflow/js/core/netflow.js b/netflow/js/core/netflow.js
index 9a88d3c..40bbbe0 100644
--- a/netflow/js/core/netflow.js
+++ b/netflow/js/core/netflow.js
@@ -43,9 +43,9 @@ function setup() {
     this.width = 0;
     this.training = null;
 
-    this.addLayer = function (n){
+    this.addLayer = function (n, activation){
         
-        var layer = new Layer(n);
+        var layer = new Layer(n, activation);
         layer.type = "OUTPUT";
         if(this.size == 0)
             layer.type = "INPUT";