Update random_subset.py