Added tests for nodes and edges, fleshed out the models a bit more