Add edge removal tests