Graph Revision

Added some graph revision from the work I did at uni and some
adjustments for the tree in Go.