Update battery_life.py