Update river.py