Update stream_avg.py