Update get_stream_samples.py