diff --git a/Generator/BaseBotGenerator.cs b/Generator/BaseBotGenerator.cs
index 2666afe..e0e7a01 100644
--- a/Generator/BaseBotGenerator.cs
+++ b/Generator/BaseBotGenerator.cs
@@ -14,12 +14,14 @@ namespace Generator
     {
         private readonly List<Datum> _rawParsedBots;
         private readonly string _workingPath;
+        private readonly string[] _botTypes;
 
         //TODO: pass in bot types and use those to create the clases in rawBots list
-        public BaseBotGenerator(List<Datum> parsedBots, string workingPath)
+        public BaseBotGenerator(List<Datum> parsedBots, string workingPath, string[] botTypes)
         {
             _rawParsedBots = parsedBots;
             _workingPath = workingPath;
+            _botTypes = botTypes;
         }
 
         public List<Bot> AddBaseDetails()
diff --git a/Generator/BotParser.cs b/Generator/BotParser.cs
index 7ae53eb..7f8d43b 100644
--- a/Generator/BotParser.cs
+++ b/Generator/BotParser.cs
@@ -15,7 +15,7 @@ namespace Generator
     {
         private readonly string _dumpPath;
 
-        public BotParser(string dumpPath, string[] botTypes)
+        public BotParser(string dumpPath)
         {
             _dumpPath = dumpPath;
         }
diff --git a/Generator/Program.cs b/Generator/Program.cs
index 1143a37..b6b8115 100644
--- a/Generator/Program.cs
+++ b/Generator/Program.cs
@@ -16,7 +16,7 @@ namespace Generator
                 "bosskilla",
                 "bosskojaniy",
                 "bosssanitar",
-                "bossstormtrooper",
+                //"bossstormtrooper",
 
                 "followerbully",
                 "followergluharassault",
@@ -25,7 +25,7 @@ namespace Generator
                 "followergluharsnipe",
                 "followerkojaniy",
                 "followersanitar",
-                "followerstormtrooper",
+                //"followerstormtrooper",
 
                 "cursedassault",
 
@@ -36,7 +36,8 @@ namespace Generator
             // Read raw bot dumps from live and turn into c# objects
             var workingPath = Directory.GetCurrentDirectory();
             var dumpPath = $"{workingPath}//dumps";
-            var botParser = new BotParser(dumpPath, botTypes);
+
+            var botParser = new BotParser(dumpPath);
             var parsedBots = botParser.Parse();
 
             if (parsedBots.Count == 0)
@@ -47,7 +48,7 @@ namespace Generator
             }
 
             // Generate the base bot class and add basic details (health/body part hp etc)
-            var baseBotGenerator = new BaseBotGenerator(parsedBots, workingPath);
+            var baseBotGenerator = new BaseBotGenerator(parsedBots, workingPath, botTypes);
             var baseBots = baseBotGenerator.AddBaseDetails();
 
             // Add weapons/armor to bot