Started adding edge class