Added set_direct_sampling