Better abstraction.
Changed to public method
Duplicate of feature/entity_distance.py