diff --git a/source/src/neurodesign.py b/source/src/neurodesign.py
index f25a7b8..86a158a 100644
--- a/source/src/neurodesign.py
+++ b/source/src/neurodesign.py
@@ -819,7 +819,7 @@ def to_next_generation(self, weights=None, seed=1234,optimisation=None):
         :type weights: list of floats, summing to 1
         :param seed: The seed for random processes.
         :type seed: integer or None
-        :param optimisation: The type of optimisation - 'GA' or 'random'
+        :param optimisation: The type of optimisation - 'GA' or 'simulation'
         :type optimisation: string
         '''
 
@@ -838,7 +838,7 @@ def to_next_generation(self, weights=None, seed=1234,optimisation=None):
             self._crossover(weights,seed)
             self._immigration(weights,noim=self.I)
 
-        elif optimisation == 'random':
+        elif optimisation == 'simulation':
             self._immigration(weights,noim=self.I)
 
         else:
@@ -886,9 +886,9 @@ def clear(self):
 
         return self
 
-    def naturalselection(self,optimisation='GA'):
+    def optimise(self,optimisation='GA'):
         '''
-        Function to run natural selection for design optimization
+        Function to run design optimization
         '''
 
         if (self.exp.FcMax == 1 and self.exp.FfMax == 1):