Create battery.py