Edited entity.py