diff --git a/src/main.py b/src/main.py
index 5869027..554b150 100644
--- a/src/main.py
+++ b/src/main.py
@@ -24,7 +24,8 @@ for i in range(0, args.iter):
     print("Epoch: %d " % i)
     pop.advance_generation()
     best = pop.best_fitness()
-    best_fit.append(pop.best_fitness().fitness())
+    best_fit.append(pop.avg_fitness())
+    #input("...")
     sleep(args.wait)
 
 plt.plot(best_fit)
diff --git a/src/population.py b/src/population.py
index 6dc17d3..19a9ebf 100644
--- a/src/population.py
+++ b/src/population.py
@@ -52,6 +52,7 @@ class Population(object):
         for member in self.members:
             position -= member.fitness() / divisor
             if position <= 0:
+                #print(member)
                 return member
 
     def mutate(self, chance):
@@ -60,6 +61,8 @@ class Population(object):
                 member.mutate()
 
     def advance_generation(self):
+        # todo: elitism
+        # todo: don't crossover all
         self.fitness_function()
         parents = list()
         for i in range(0, len(self.members)):
@@ -71,4 +74,4 @@ class Population(object):
             children.append(one)
             children.append(two)
         self.members = children
-        self.mutate(5)
+        self.mutate(2)