Fitness function

Add the fitness function