improved the widget