Added sending indexes of variables instead of masks in predict mode