Almost finished including edge-wise feature probability, but something is very wrong in the adjust/recalculate methods