added the edge class fixed the node class