Improve widget for battery