Add battery.py