Reorganize unit tests
Rename AddNode method to AddEdge. Other minor changes.