add battery support