fitness function improvements