add edges back in