Add lambd_weight function