added fitness function code