Fixed unlabeled edge issue. Still trying to separate out all test code.