Moved around edge comparison to actually work