Added code for the edge class