Add edge class and fixed test_edge