Added code to properly display connections between entities.