code now uses correct fitness