Refacotr edge class: