Removed the requirement to explicity call add_entity. It's now called in Entity.__init__()