adding comment to entities.py and fixing small bug.