ENH: add edge weights