Update HW2.py

Updated for calculation of mean annual discharge