Update reservoir_sampling.py