diff --git a/config.txt b/config.txt
index 4d036e0..a1fc871 100644
--- a/config.txt
+++ b/config.txt
@@ -1,14 +1,14 @@
 [NEAT]
-fitness_criterion     = max
-fitness_threshold     = 3.9
-pop_size              = 150
+fitness_criterion     = mean
+fitness_threshold     = 400
+pop_size              = 1000
 reset_on_extinction   = False
 
 [DefaultGenome]
 # node activation options
-activation_default      = sigmoid
-activation_mutate_rate  = 0.0
-activation_options      = sigmoid
+activation_default      = relu
+activation_mutate_rate  = 1.0
+activation_options      = relu
 
 # node aggregation options
 aggregation_default     = sum
@@ -37,14 +37,14 @@ enabled_default         = True
 enabled_mutate_rate     = 0.01
 
 feed_forward            = True
-initial_connection      = full
+initial_connection      = full_direct
 
 # node add/remove rates
 node_add_prob           = 0.2
 node_delete_prob        = 0.2
 
 # network parameters
-num_hidden              = 0
+num_hidden              = 2
 num_inputs              = 17
 num_outputs             = 4
 
diff --git a/main.py b/main.py
index 0fe8836..2ebf78a 100755
--- a/main.py
+++ b/main.py
@@ -26,13 +26,15 @@ group.add_argument(
     action="store_true",
     help="Debug",
 )
+
 group.add_argument(
     "-v",
     "--verbose",
     action="store_true",
     help="Verbose",
 )
-group.add_argument(
+
+parser.add_argument(
     "-t",
     "--train",
     type=pos_int,
@@ -63,6 +65,7 @@ def main(args: argparse.ArgumentParser) -> None:
         setup_logger("warning")
 
     if args.train is not None:
+        logger.warning(args.train)
         train(args.train)
     else:
         Menu().run()