Added removeEdge, added more tests