Update entity.py