FTAG: Add comments to GNN code & cleanup

Samuel Van Stroudrequested to merge
svanstro/athena:svanstro/add-comments-to-ftag-gnn into main

I was looking for a bug so added some comments. Fortunately there was no bug. But now there are comments!

Merge request reports