update and fix for battery