Fix associations, User model and messages controller for widget