diff --git a/src/EasyGA.py b/src/EasyGA.py
index 2a9cc57..bd9b2a3 100644
--- a/src/EasyGA.py
+++ b/src/EasyGA.py
@@ -2,13 +2,14 @@
 from initialization.random_initialization import random_initialization
 
 import random
-from defaults import defaults
-import gene
+from defaults.defaults import defaults
+from gene.gene import gene
 
 class GA:
     def __init__(self):
         # Default variables
-         self.gene = defaults.default_gene_function()
+         self.gene = gene(defaults.default_gene_function())
+         self.chromosome_length = defaults.chromosome_length
     #     self.population_size = defaults.generations
     #     self.chromosome_length = defaults.chromosome_length
     #     self.generations = defaults.generations
diff --git a/src/run_testing.py b/src/run_testing.py
index 9e2b099..7aee699 100644
--- a/src/run_testing.py
+++ b/src/run_testing.py
@@ -3,3 +3,6 @@ import EasyGA
 
 # Create the Genetic algorithm
 ga = EasyGA.GA()
+
+print(ga.gene.get_fitness())
+print(ga.gene.get_value())