Add user id to model