Imported from https://github.com/alextanhongpin/affinity-analysis