Update riverHeight.py