Hey, I don't know how active the HNN project is, but after looking at your code draft, I have a few suggestions. You guys are making it way too hard on yourselves by going to a graph representation. This will no doubt have a performance impact and will / already has introduced needless complexity to your codes.