Update tree and edge classes
Update test
Update tests
Update edge class
Update edge class