Starting to add support for dealing with edges
Fixed bug in retrieving edges and improves tests