Added in add_edge functionality.