Adding the weighted edge function