Update battery.py