Update discharge.py