Update egonet.py