Add battery status feature