add battery class