Wrote more tests for battery class