Update __init__.py

Added EntityRelationship to the available models