Add battery capacity info
Closes #8