Added some functionality to batterytest