Fixes Affinity method name
Adds affinity model specs