Add sampling of the weights vector