Added ability to remove edges