Thanks a lot for your work, It is helping everyone a lot. I was wondering, if I want to replace EGCNN with GCN, then what changes do I need to do?