Add WidgetManager class