Update edge class
Update edge
Fix