Fixed: add data for edges correctly