Create battery_life.py