In pump_electricity_zo.py, most of the variables are not indexed by time. They should be updated so they are indexed by time.