Add updateWidget functionality