Wrote first test for Battery class