Improve UserWidget