Adding affinity score