Update entities.py