Fix battery.lua bug
Add stuff