FTAG: Add comments to GNN code & cleanup
I was looking for a bug so added some comments. Fortunately there was no bug. But now there are comments!
I was looking for a bug so added some comments. Fortunately there was no bug. But now there are comments!