Add UserWidget class