WIP: Add labels for each edge, must recheck edge labels