change random.sample